package com.android.utils.rest;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.URLEncoder;
import java.util.ArrayList;

import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.NameValuePair;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.HttpClient;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpUriRequest;
import org.apache.http.conn.params.ConnManagerParams;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.message.BasicNameValuePair;
import org.apache.http.params.HttpConnectionParams;
import org.apache.http.params.HttpParams;
import org.apache.http.protocol.HTTP;

import android.util.Base64;
import android.util.Log;

public class RestClient
{
	private ArrayList<NameValuePair> params;
	private ArrayList<NameValuePair> headers;
	
	private int responseCode;
	
	private String url;
	private String message;
	private String response;
	
	public static final String AUTHORIZATION = "Authorization";
	
	private Integer registrationTimeOut = null;
	private Integer waitTimeOut = null;
	
	public enum RequestMethod
	{
		GET, POST;
	}

	public RestClient(String url)
	{
		this.url = url;
		params = new ArrayList<NameValuePair>();
		headers = new ArrayList<NameValuePair>();
	}
	
	public String getResponse()
	{
		return response;
	}

	public String getErrorMessage()
	{
		return message;
	}

	public int getResponseCode()
	{
		return responseCode;
	}
	
	public void setTimeValues(Integer registrationTimeOut, Integer waitTimeOut) 
	{
		this.registrationTimeOut = registrationTimeOut;
		this.waitTimeOut = waitTimeOut;
	}
	
	public void addParam(String name, String value)
	{
		params.add(new BasicNameValuePair(name, value));
	}

	public void addHeader(String name, String value)
	{
		headers.add(new BasicNameValuePair(name, value));
	}
	
	public void setBasicAuthorization(String key, String login, String password)
	{
		headers.add(new BasicNameValuePair(key, "Basic " + new String(Base64.encode((login + ":" + password).getBytes(), Base64.NO_WRAP))));
	}

	public void execute(RequestMethod method) throws Exception
	{
		switch (method)
		{
			case GET:
			{
				String combinedParams = "";
				if (!params.isEmpty())
				{
					combinedParams += "?";
					for (NameValuePair p : params)
					{
						String paramString = p.getName() + "=" + URLEncoder.encode(p.getValue(), HTTP.UTF_8);
						if (combinedParams.length() > 1)
						{
							combinedParams += "&" + paramString;
						}
						else
						{
							combinedParams += paramString;
						}
					}
				}
	
				HttpGet request = new HttpGet(url + combinedParams);
	
				for (NameValuePair h : headers)
				{
					request.addHeader(h.getName(), h.getValue());
				}
	
				executeRequest(request, url);
				break;
			}
			case POST:
			{
				HttpPost request = new HttpPost(url);
				
				for (NameValuePair h : headers)
				{
					request.addHeader(h.getName(), h.getValue());
				}
	
				if (!params.isEmpty())
				{
					request.setEntity(new UrlEncodedFormEntity(params, HTTP.UTF_8));
				}
	
				executeRequest(request, url);
				break;
			}
		}
	}

	private void executeRequest(HttpUriRequest request, String url)
	{
		
		HttpClient client = new DefaultHttpClient();
		
		if(registrationTimeOut != null) 
		{
			HttpParams settings = client.getParams();
			HttpConnectionParams.setConnectionTimeout(settings, registrationTimeOut);
			HttpConnectionParams.setSoTimeout(settings, waitTimeOut);
			ConnManagerParams.setTimeout(settings, waitTimeOut);
			request.setParams(settings);
		}

		HttpResponse httpResponse;

		try
		{
			httpResponse = client.execute(request);
			responseCode = httpResponse.getStatusLine().getStatusCode();
			message = httpResponse.getStatusLine().getReasonPhrase();

			HttpEntity entity = httpResponse.getEntity();

			if (entity != null)
			{

				InputStream instream = entity.getContent();
				response = convertStreamToString(instream);

				instream.close();
			}

		}
		catch (ClientProtocolException e)
		{
			client.getConnectionManager().shutdown();
			e.printStackTrace();
		}
		catch (IOException e)
		{
			client.getConnectionManager().shutdown();
			e.printStackTrace();
		}
	}

	private static String convertStreamToString(InputStream is)
	{
		String line = "";
		StringBuilder total = new StringBuilder();

		BufferedReader rd = new BufferedReader(new InputStreamReader(is));

		try
		{
			while ((line = rd.readLine()) != null)
			{
				total.append(line);
			}
		}
		catch (IOException e)
		{
			Log.e("RestClient", e.getLocalizedMessage(), e);
		}

		return total.toString();
	}
}
